//
//  UIWindow+CustomDatePicker.m
//  DrivieSchoolStudents
//
//  Created by TangBin on 2019/6/19.
//  Copyright © 2019 班文政. All rights reserved.
//

#import "UIWindow+CustomDatePicker.h"
#import <objc/runtime.h>

@implementation UIWindow (CustomDatePicker)


- (void)setCustDatePicker:(EDSCustomDatePicker *)custDatePicker
{
    objc_setAssociatedObject(self, @selector(custDatePicker), custDatePicker, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (EDSCustomDatePicker *)custDatePicker
{
    return objc_getAssociatedObject(self, _cmd);
}

- (void)setCustDatePickerMaskView:(UIView *)custDatePickerMaskView
{
    objc_setAssociatedObject(self, @selector(custDatePickerMaskView), custDatePickerMaskView, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (UIView *)custDatePickerMaskView
{
    return objc_getAssociatedObject(self, _cmd);
}

@end
